/**
 * Copyright (c) 2013-2020 Contributors to the Eclipse Foundation
 *
 * <p> See the NOTICE file distributed with this work for additional information regarding copyright
 * ownership. All rights reserved. This program and the accompanying materials are made available
 * under the terms of the Apache License, Version 2.0 which accompanies this distribution and is
 * available at http://www.apache.org/licenses/LICENSE-2.0.txt
 */
package org.locationtech.geowave.analytic.mapreduce;

import java.io.IOException;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.mapreduce.Counters;
import org.apache.hadoop.mapreduce.Job;
import org.apache.hadoop.util.Tool;
import org.apache.hadoop.util.ToolRunner;
import org.locationtech.geowave.analytic.PropertyManagement;

public class ToolRunnerMapReduceIntegration implements MapReduceIntegration {

  @Override
  public Job getJob(final Tool tool) throws IOException {
    return new Job(tool.getConf());
  }

  @Override
  public int submit(
      final Configuration configuration,
      final PropertyManagement runTimeProperties,
      final GeoWaveAnalyticJobRunner tool) throws Exception {
    return ToolRunner.run(configuration, tool, new String[] {});
  }

  @Override
  public Counters waitForCompletion(final Job job)
      throws ClassNotFoundException, InterruptedException, Exception {
    final boolean status = job.waitForCompletion(true);
    return status ? job.getCounters() : null;
  }

  @Override
  public Configuration getConfiguration(final PropertyManagement runTimeProperties)
      throws IOException {
    return MapReduceJobController.getConfiguration(runTimeProperties);
  }
}
